// 实现异步缓冲区
#ifndef __M_BUFFER_H__
#define __M_BUFFER_H__
#include "util.hpp"
#include <vector>
#include <cassert>

namespace log
{
#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
#define THRESHOLD_BUFFER_SIZE (8 * 1024 * 1024)
#define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024)
    class Buffer
    {
    public:
        Buffer() : _buffer(DEFAULT_BUFFER_SIZE), _writer_idx(0), _reader_idx(0) {}
        // 向缓冲区写入数据
        void push(const char *data, size_t len)
        {
            // 考虑空间不够-扩容
            ensureEnoughSize(len);
            // 1. 将数据拷贝至缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 2. 将当前写入位置向后移动
            moveWriter(len);
        }
        // 返回可读数据的起始地址
        const char *begin()
        {
            return &_buffer[_reader_idx];
        }
        // 返回可读数据的长度,也就是缓冲区实际存在数据的长度
        size_t readAbleSize()
        {
            return (_writer_idx - _reader_idx);
        }
        // 剩余可写空间大小
        size_t writeAbleSize()
        {
            // 对于扩容来说,无需要这个接口，因为总是有剩余空间可写，因为插入数据时已经扩容过了
            // 这个接口针对规定大小提供
            return (_buffer.size() - _writer_idx);
        }
        // 对读写指针进行向后移动
        void moveWriter(size_t len)
        {
            assert(len + _writer_idx <= _buffer.size());
            _writer_idx += len;
        }
        void moveReader(size_t len)
        {
            assert(len <= readAbleSize());
            _reader_idx += len;
        }
        // 重置读写位置，初始化缓冲区,缓冲区不存在可读条件，但存在可写条件
        void reset()
        {
            _writer_idx = 0;
            _reader_idx = 0;
        }
        // 对buffer进行交换操作
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return _writer_idx == _reader_idx;
        }

    private:
        // 对空间进行扩容
        void ensureEnoughSize(size_t len)
        {
            if (len <= writeAbleSize())
                return; // 无需扩容，因为剩余可写空间足够该len长度数据插入
            size_t new_size = 0;
            if (_buffer.size() < THRESHOLD_BUFFER_SIZE)
            {
                new_size = _buffer.size() * 2 + len; // 翻倍增长
            }
            else
            {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len; // 线性增长
            }

            _buffer.resize(new_size);
        }

    private:
        std::vector<char> _buffer;
        size_t _reader_idx; // 当前指向可读的指针
        size_t _writer_idx; // 当前指向可写的指针
    };
}

#endif